#include <stdio.h>
#include "../base_lib/CryptoHelper.h"
#include <string>
#include <stdlib.h>
#include <iostream> 

using namespace std;
namespace license {

void write_pubkey_file(string public_fname, BYTE *pbPublicKey,
		DWORD dwPublicKeySize) {
	FILE* fp = fopen(public_fname.c_str(), "w");
	fprintf(fp, "//file generated by bootstrap.cpp, do not edit.\n\n");
	fprintf(fp, "#ifndef PUBLIC_KEY_H_\n#define PUBLIC_KEY_H_\n");
	fprintf(fp, "static BYTE PUBLIC_KEY[] = {");
	for (int i = 0; i < dwPublicKeySize; i++) {
		if (i != 0) {
			fprintf(fp, ",");
		}
		if (i % 15 == 0) {
			fprintf(fp, "\n        ");
		}
		fprintf(fp, "%d", pbPublicKey[i]);
	}
	fprintf(fp, "\n};\n\n");
	int random = rand() % 1000;
	fprintf(fp, "#define SHARED_RANDOM %d;\n", random);
	fprintf(fp, "#endif\n");
	fclose(fp);
}

void write_privkey_file(string private_fname, BYTE *privateKey,
		DWORD dwPrivateKeySize) {
	FILE* fp = fopen(private_fname.c_str(), "w");
	fprintf(fp, "//file generated by bootstrap.cpp, do not edit.\n\n");
	fprintf(fp, "#ifndef PRIVATE_KEY_H_\n#define PRIVATE_KEY_H_\n");
	fprintf(fp, "static BYTE PRIVATE_KEY[] = {");
	for (int i = 0; i < dwPrivateKeySize; i++) {
		if (i != 0) {
			fprintf(fp, ",");
		}
		if (i % 15 == 0) {
			fprintf(fp, "\n        ");
		}
		fprintf(fp, "%d", privateKey[i]);
	}
	fprintf(fp, "\n};\n\n");
	fprintf(fp, "#endif\n");
	fclose(fp);
}

void generatePk(string private_include, string public_include) {
	unique_ptr<CryptoHelper> cryptoHlpr = CryptoHelper.getInstance();
	BYTE *pbPublicKey = NULL, *pbPrivateKey = NULL;
	DWORD dwPublicKeySize = 0, dwPrivateKeySize = 0;
	HRESULT hr = S_OK;
	// Get the key container context.
	if (FAILED(hr = cryptoHlpr.AcquireContext(_T("TestContainer")))) {
// Call FormatMessage to display the error returned in hr.
		return;
	}
	// Generate the public/private key pair.
	if (FAILED(hr = cryptoHlpr.GenerateKeyPair())) {
// Call FormatMessage to display the error returned in hr.
		return;
	}
	// Export out the public key blob.
	if (FAILED(
			hr = cryptoHlpr.ExportPublicKey(&pbPublicKey, dwPublicKeySize))) {
// Call FormatMessage to display the error returned in hr.
		cerr << "error exporting pubkey" << endl;
		return;
	} else {
		write_pubkey_file(public_include, pbPublicKey, dwPublicKeySize);
	}
	// Print out the public key to console as a
	// hexadecimal string.
	wprintf(L"\n\nPublicKey = \"");
	for (DWORD i = 0; i < dwPublicKeySize; i++) {
		wprintf(L"%02x", pbPublicKey[i]);
	}
	wprintf(L"\"\n");
	// Export out the private key blob.
	if (FAILED(cryptoHlpr.ExportPrivateKey(&pbPrivateKey, dwPrivateKeySize))) {
		cerr << "Error exporting private key." << endl;
		return;
	} else {
		write_privkey_file(private_include, pbPrivateKey, dwPrivateKeySize);
	}
	// Print out the private key to console as a
	// hexadecimal string.
	wprintf(L"\n\nPrivateKey = \"");
	for (DWORD i = 0; i < dwPrivateKeySize; i++) {
		wprintf(L"%02x", pbPrivateKey[i]);
	}
	wprintf(L"\"\n");
	// Delete the public key blob allocated by the
// ExportPublicKey method.
	if (pbPublicKey)
		delete[] pbPublicKey;
	// Delete the private key blob allocated by the
// ExportPrivateKey method.
	if (pbPrivateKey)
		delete[] pbPrivateKey;
	return;
}
}

int main(int argc, char** argv) {

	if (argc != 3) {
		//print_usage();
		exit(2);
	} else {
		printf("********************************************\n");
		printf("*  Bootstrap!!!                            *\n");
		printf("********************************************\n");

	}
	string private_fname = string(argv[1]);
	string public_fname(argv[2]);

	license::generatePk(private_fname, public_fname);
	return 0;
}